AI-assisted models providing diagnosis and prognostic information